package br.com.stefanini.schs.util;

import java.util.Date;
import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

import br.com.stefanini.schs.ejb.entity.Ponto;

public class ListaPontoModelo extends ListDataModel<Ponto> implements SelectableDataModel<Ponto> {

	public ListaPontoModelo() {}
	
	public ListaPontoModelo(List<Ponto> data) {
		super(data);
	}
	
	@Override
	@SuppressWarnings("unchecked")
	public Ponto getRowData(String rowKey) {

		List<Ponto> listaPonto = (List<Ponto>) getWrappedData();
		
		for (Ponto ponto : listaPonto) {
			if (ponto.getFuncionario().getEmail().equals(rowKey))
				return ponto;
		}
		return null;
	}
	
	public Date getRowKey(Date data) {
		return data;
	}
	
	@SuppressWarnings("unchecked")
	public Ponto getRowData(long idPonto) {

		List<Ponto> listaPontos = (List<Ponto>) getWrappedData();
		
		for (Ponto ponto : listaPontos) {
			if (ponto.getIdPonto() == idPonto)
				return ponto;
		}
		return null;
	}

	@Override
	public Object getRowKey(Ponto arg0) {
		return null;
	}
}
